#include "flutter_screen_recording_plugin.h"

// This must be included before many other Windows headers.
#include <windows.h>

// For getPlatformVersion; remove unless needed for your plugin implementation.
#include <VersionHelpers.h>

#include <flutter/method_channel.h>
#include <flutter/plugin_registrar_windows.h>
#include <flutter/standard_method_codec.h>

#include <memory>
#include <sstream>
#include "RecordInterface.h"

static std::string record_file_path;

namespace flutter_screen_recording {

// static
void FlutterScreenRecordingPlugin::RegisterWithRegistrar(
    flutter::PluginRegistrarWindows *registrar) {
  auto channel =
      std::make_unique<flutter::MethodChannel<flutter::EncodableValue>>(
          registrar->messenger(), "flutter_screen_recording",
          &flutter::StandardMethodCodec::GetInstance());

  auto plugin = std::make_unique<FlutterScreenRecordingPlugin>();

  channel->SetMethodCallHandler(
      [plugin_pointer = plugin.get()](const auto &call, auto result) {
        plugin_pointer->HandleMethodCall(call, std::move(result));
      });

  registrar->AddPlugin(std::move(plugin));
}

FlutterScreenRecordingPlugin::FlutterScreenRecordingPlugin() {}

FlutterScreenRecordingPlugin::~FlutterScreenRecordingPlugin() {}

void FlutterScreenRecordingPlugin::HandleMethodCall(
    const flutter::MethodCall<flutter::EncodableValue> &method_call,
    std::unique_ptr<flutter::MethodResult<flutter::EncodableValue>> result) {
  if (method_call.method_name().compare("getPlatformVersion") == 0) {
    std::ostringstream version_stream;
    version_stream << "Windows ";
    if (IsWindows10OrGreater()) {
      version_stream << "10+";
    } else if (IsWindows8OrGreater()) {
      version_stream << "8";
    } else if (IsWindows7OrGreater()) {
      version_stream << "7";
    }
    result->Success(flutter::EncodableValue(version_stream.str()));
  }
  else if (method_call.method_name().compare("startRecordScreen") == 0) {
        auto* arguments = std::get_if<flutter::EncodableMap>(method_call.arguments());
        std::string path = std::get<std::string>(arguments->at(flutter::EncodableValue("path")));
        record_file_path = path;
        bool audio = std::get<bool>(arguments->at(flutter::EncodableValue("audio")));
        startRecord(path.c_str(),audio);
        result->Success(true);
  }
  else if(method_call.method_name().compare("stopRecordScreen") == 0){
     stopRecord();
     result->Success(record_file_path);
  }
   else {
    result->NotImplemented();
  }
}

}  // namespace flutter_screen_recording
